# File   : raw.py
# Author : Zhengkun Tian
# Email  : zhengkun.tian@outlook.com

import oneflow as torch
import oneflow.nn as nn
from oasr.frontend.base import BaseFrontEnd


class ConvFrontFromRAW(BaseFrontEnd):
    def __init__(self, in_channels=1, output_size=40, num_conv_layers=5, out_frame_rate=30, dropout=0.0):
        super(ConvFrontFromRAW, self).__init__()


    def forward(self, inputs, inputs_mask):
        """
        Args:
            inputs: [batch_size, channels, time1]
            inputs_mask: [batch_size, 1, time1]
        Returns:
            tensor: [batch_size, time2, dim]
            tensor_mask: [batch_size, time2, dim]
        """

        return inputs, inputs_mask

    def inference(self, inputs, inputs_mask, cache=None):
        raise NotImplementedError